<?php
/**
 * Zend Framework
 *
 * LICENSE
 *
 * This source file is subject to the new BSD license that is bundled
 * with this package in the file LICENSE.txt.
 * It is also available through the world-wide-web at this URL:
 * http://framework.zend.com/license/new-bsd
 * If you did not receive a copy of the license and are unable to
 * obtain it through the world-wide-web, please send an email
 * to license@zend.com so we can send you a copy immediately.
 *
 * @category   Zend
 * @package    Zend_Test
 * @subpackage PHPUnit
 * @copyright  Copyright (c) 2005-2009 Zend Technologies USA Inc. (http://www.zend.com)
 * @license    http://framework.zend.com/license/new-bsd     New BSD License
 * @version    $Id: Truncate.php 19106 2009-11-20 17:15:30Z beberlei $
 */

require_once "PHPUnit/Extensions/Database/Operation/IDatabaseOperation.php";

require_once "PHPUnit/Extensions/Database/DB/IDatabaseConnection.php";

require_once "PHPUnit/Extensions/Database/DataSet/IDataSet.php";

require_once "PHPUnit/Extensions/Database/Operation/Exception.php";

/**
 * @see Zend_Test_PHPUnit_Db_Connection
 */
require_once "Zend/Test/PHPUnit/Db/Connection.php";

/**
 * Operation for Truncating on setup or teardown of a database tester.
 *
 * @uses       PHPUnit_Extensions_Database_Operation_IDatabaseOperation
 * @category   Zend
 * @package    Zend_Test
 * @subpackage PHPUnit
 * @copyright  Copyright (c) 2005-2009 Zend Technologies USA Inc. (http://www.zend.com)
 * @license    http://framework.zend.com/license/new-bsd     New BSD License
 */
class Zend_Test_PHPUnit_Db_Operation_Truncate implements PHPUnit_Extensions_Database_Operation_IDatabaseOperation
{
	/**
	 *
	 * @param PHPUnit_Extensions_Database_DB_IDatabaseConnection $connection
	 * @param PHPUnit_Extensions_Database_DataSet_IDataSet $dataSet
	 * @return void
	 */
	public function execute(PHPUnit_Extensions_Database_DB_IDatabaseConnection $connection, PHPUnit_Extensions_Database_DataSet_IDataSet $dataSet)
	{
		if(!($connection instanceof Zend_Test_PHPUnit_Db_Connection)) {
			require_once "Zend/Test/PHPUnit/Db/Exception.php";
			throw new Zend_Test_PHPUnit_Db_Exception("Not a valid Zend_Test_PHPUnit_Db_Connection instance, ".get_class($connection)." given!");
		}

		foreach ($dataSet->getReverseIterator() AS $table) {
			try {
				$tableName = $table->getTableMetaData()->getTableName();
				$this->_truncate($connection->getConnection(), $tableName);
			} catch (Exception $e) {
				throw new PHPUnit_Extensions_Database_Operation_Exception('TRUNCATE', 'TRUNCATE '.$tableName.'', array(), $table, $e->getMessage());
			}
		}
	}

	/**
	 * Truncate a given table.
	 *
	 * @param Zend_Db_Adapter_Abstract $db
	 * @param string $tableName
	 * @return void
	 */
	protected function _truncate(Zend_Db_Adapter_Abstract $db, $tableName)
	{
		$tableName = $db->quoteIdentifier($tableName);
		if($db instanceof Zend_Db_Adapter_Pdo_Sqlite) {
			$db->query('DELETE FROM '.$tableName);
		} else if($db instanceof Zend_Db_Adapter_Db2) {
			/*if(strstr(PHP_OS, "WIN")) {
			 $file = tempnam(sys_get_temp_dir(), "zendtestdbibm_");
			 file_put_contents($file, "");
			 $db->query('IMPORT FROM '.$file.' OF DEL REPLACE INTO '.$tableName);
			 unlink($file);
			 } else {
			 $db->query('IMPORT FROM /dev/null OF DEL REPLACE INTO '.$tableName);
			 }*/
			require_once "Zend/Exception.php";
			throw Zend_Exception("IBM Db2 TRUNCATE not supported.");
		} else if($this->_isMssqlOrOracle($db)) {
			$db->query('TRUNCATE TABLE '.$tableName);
		} else if($db instanceof Zend_Db_Adapter_Pdo_Pgsql) {
			$db->query('TRUNCATE '.$tableName.' CASCADE');
		} else {
			$db->query('TRUNCATE '.$tableName);
		}
	}

	/**
	 * Detect if an adapter is for Mssql or Oracle Databases.
	 *
	 * @param  Zend_Db_Adapter_Abstract $db
	 * @return bool
	 */
	private function _isMssqlOrOracle($db)
	{
		return (
		$db instanceof Zend_Db_Adapter_Pdo_Mssql ||
		$db instanceof Zend_Db_Adapter_Sqlsrv ||
		$db instanceof Zend_Db_Adapter_Pdo_Oci ||
		$db instanceof Zend_Db_Adapter_Oracle
		);
	}
}